# 创建大语言模型
from langchain_core.prompts import ChatPromptTemplate

from base import model

# 输入提示词模板
# prompt = [
#     SystemMessage(content="你是一个用于Java开发的智能代码助手"),
#     HumanMessage(content="写一个快速排序方法")
# ]
prompt = ChatPromptTemplate.from_messages(
    [
        ("system","你是一个用于{lang}开发的智能代码助手"),
        ("user","{input}"),
    ]
)
print(prompt)
# chain = prompt.format({"input":"写一个快速排序算法"}) | llm

# 执行大模型计算 获取 结果
chain = prompt | model

modelResult = chain.invoke({'lang':'Java','input':'写一个hello world程序'})
print(f"-----model-result----- \n${modelResult}")




